Image Classification

This article is based on the TensorFlow Image Classification article where we demonstrate image classification using TensorFlow. The dataset that we use here is a filtered version of Dogs vs Cats dataset from Kaggle.

Dogs vs. Cats Dataset

Downloading Data

Dataset Directory Structure

Preprocessing

Train Data Image DataGenerator

Validation Data Image DataGenerator

Modeling

We use Keras Sequential model for creating a model.

Compiling and fitting the model


References

  1. TensorFlow Image
  2. Image classification
  3. Dogs vs. Cats